#pragma once
#include <iostream>
#include <vector>

#include "Mutex.hpp"
#include "Sem.hpp"

namespace RingBufferModule
{
    using namespace SemModule;
    using namespace LockModule;
    template <class T>
    class RingBuffer
    {
    public:
        RingBuffer(int cap)
            : _cap(cap), _ring(cap), _c_step(0), _p_step(0), _datasem(0), _spacesem(cap)
        {
        }
        void Equeue(const T &in)
        {
            // 生产者
            _spacesem.P(); // 先申请信号量再排队

            {
                LockGuard lockguard(_p_lock);
                _ring[_p_step] = in; // 生产完毕
                _p_step++;
                _p_step %= _cap; // 维持环形特性
            }

            _datasem.V();
        }
        void Pop(T *out)
        {
            // 消费者
            _datasem.P();

            {
                LockGuard lockguard(_c_lock);
                *out = _ring[_c_step];
                _c_step++;
                _c_step %= _cap;
            }

            _spacesem.V();
        }
        ~RingBuffer()
        {
        }

    private:
        std::vector<T> _ring; // 环(临界资源)
        int _cap;             // 总容量
        int _c_step;          // 消费者位置
        int _p_step;          // 生产者位置

        Sem _datasem;  // 数据信号量
        Sem _spacesem; // 空间信号量

        Mutex _p_lock; // 生产者之间的锁
        Mutex _c_lock; // 消费者之间的锁
    };
}